package com.qinglei.recoup.longconn.service;

import com.alibaba.fastjson.JSON;
import com.aliyuncs.DefaultAcsClient;
import com.aliyuncs.IAcsClient;
import com.aliyuncs.exceptions.ClientException;
import com.aliyuncs.exceptions.ServerException;
import com.aliyuncs.iot.model.v20180120.ResetConsumerGroupPositionRequest;
import com.aliyuncs.iot.model.v20180120.ResetConsumerGroupPositionResponse;
import com.aliyuncs.profile.DefaultProfile;
import com.google.gson.Gson;
import com.qinglei.recoup.common.domain.IotRadarStatusData;
import com.qinglei.recoup.common.domain.RadarMessage;
import com.qinglei.recoup.longconn.properties.IotProperties;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.binary.Base64;
import org.apache.commons.lang3.StringUtils;
import org.apache.qpid.jms.JmsConnection;
import org.apache.qpid.jms.JmsConnectionListener;
import org.apache.qpid.jms.message.JmsInboundMessageDispatch;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import javax.jms.*;
import javax.naming.Context;
import javax.naming.InitialContext;
import java.net.URI;
import java.util.HashMap;
import java.util.Hashtable;
import java.util.Map;
import java.util.Objects;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;

/**
 * 物联网消息对接服务
 */
@Service
@Slf4j   //
public class IotMessageService implements InitializingBean {

    @Resource
    private IotProperties iotProperties;

    @Resource
    private MessageProcService messageProcService;

    //业务处理异步线程池，线程池参数可以根据您的业务特点调整，或者您也可以用其他异步方式处理接收到的消息。
    private final ExecutorService executorService = new ThreadPoolExecutor(
            Runtime.getRuntime().availableProcessors(),
            Runtime.getRuntime().availableProcessors() * 2, 60, TimeUnit.SECONDS,
            new LinkedBlockingQueue<>(50000));

    @Autowired
    private IotMessageHandleService iotMessageHandleService;

    @Override
    public void afterPropertiesSet() throws Exception {
        //参数说明，请参见AMQP客户端接入说明文档。
        String accessKey = iotProperties.getAccessKey();
        String accessSecret = iotProperties.getAccessSecret();
        String consumerGroupId = iotProperties.getConsumerGroupId();
        String amqpHost = iotProperties.getAmqpHost();
        //iotInstanceId：企业版实例请填写实例ID，公共实例请填空字符串""。
        String iotInstanceId = iotProperties.getIotInstanceId();

        //清空堆积消息
        clearMessage(accessKey, accessSecret, iotInstanceId, consumerGroupId);

        long timeStamp = System.currentTimeMillis();
        //签名方法：支持hmacmd5、hmacsha1和hmacsha256。
        String signMethod = "hmacsha1";
        //控制台服务端订阅中消费组状态页客户端ID一栏将显示clientId参数。
        //建议使用机器UUID、MAC地址、IP等唯一标识等作为clientId。便于您区分识别不同的客户端。
        String clientId = iotProperties.getClientId();

        //userName组装方法，请参见AMQP客户端接入说明文档。
        String userName = clientId + "|authMode=aksign"
                + ",signMethod=" + signMethod
                + ",timestamp=" + timeStamp
                + ",authId=" + accessKey
                + ",iotInstanceId=" + iotInstanceId
                + ",consumerGroupId=" + consumerGroupId
                + "|";
        //计算签名，password组装方法，请参见AMQP客户端接入说明文档。
        String signContent = "authId=" + accessKey + "&timestamp=" + timeStamp;
        String password = doSign(signContent, accessSecret, signMethod);
        //接入域名，请参见AMQP客户端接入说明文档。
        String connectionUrl = "failover:(amqps://" + amqpHost + ":5671?amqp.idleTimeout=80000)"
                + "?failover.reconnectDelay=30";

        Hashtable<String, String> hashtable = new Hashtable<>();
        hashtable.put("connectionfactory.SBCF", connectionUrl);
        hashtable.put("queue.QUEUE", "default");
        hashtable.put(Context.INITIAL_CONTEXT_FACTORY, "org.apache.qpid.jms.jndi.JmsInitialContextFactory");
        Context context = new InitialContext(hashtable);
        ConnectionFactory cf = (ConnectionFactory) context.lookup("SBCF");
        Destination queue = (Destination) context.lookup("QUEUE");
        // 创建连接。
        Connection connection = cf.createConnection(userName, password);
        ((JmsConnection) connection).addConnectionListener(myJmsConnectionListener);
        // 创建会话。
        // Session.CLIENT_ACKNOWLEDGE: 收到消息后，需要手动调用message.acknowledge()。
        // Session.AUTO_ACKNOWLEDGE: SDK自动ACK（推荐）。
        Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
        connection.start();
        // 创建Receiver连接。
        MessageConsumer consumer = session.createConsumer(queue);
        consumer.setMessageListener(messageListener);
    }

    private MessageListener messageListener = new MessageListener() {
        @Override
        public void onMessage(javax.jms.Message message) {
            try {
                //1.收到消息之后一定要ACK。
                // 推荐做法：创建Session选择Session.AUTO_ACKNOWLEDGE，这里会自动ACK。
                // 其他做法：创建Session选择Session.CLIENT_ACKNOWLEDGE，这里一定要调message.acknowledge()来ACK。
                // message.acknowledge();
                //2.建议异步处理收到的消息，确保onMessage函数里没有耗时逻辑。
                // 如果业务处理耗时过程过长阻塞住线程，可能会影响SDK收到消息后的正常回调。
                executorService.submit(() -> processMessage(message));
            } catch (Exception e) {
                log.error("submit task occurs exception ", e);
            }
        }
    };

    /**
     * 在这里处理您收到消息后的具体业务逻辑。
     */
    private void processMessage(javax.jms.Message message) {


        try {
            byte[] body = message.getBody(byte[].class);
            String content = new String(body);
            log.info("$$$$$: 【{}】", content);
            String topic = message.getStringProperty("topic");
//            String messageId = message.getStringProperty("messageId");
//            log.info("receive message"
//                    + ", topic = " + topic
//                    + ", messageId = " + messageId
//                    + ", content = " + content);
            if (topic.endsWith("/user/update")) {
                // 处理上传灰度数据
                log.warn("Unsupported topic:{}, content:{}", topic, content);
//                String[] split = topic.split("/");
//                String sn = split[2];
//                IotDebugBody iotDebugBody = new IotDebugBody(topic, messageId, content);
//                com.qinglei.recoup.longconn.entity.Message qlMsg = new com.qinglei.recoup.longconn.entity.Message();
//                qlMsg.setFuncCode(LongConnConstant.FUNC_UP_USER_DATA);
//                qlMsg.setSn(sn);
//                qlMsg.setBody(iotDebugBody);
//                qlMsg.setTimestamp(DateUtil.formatFullTime(LocalDateTime.now(), DateUtil.FULL_TIME_MILLI_BAR_PATTERN));
//                fileService.save(qlMsg);
            } else if (topic.startsWith("/as/mqtt/status")) {
                // 处理设备上下线消息
                log.info("设备上下线消息process topic:{}, content:{}", topic, content);
                String[] split = topic.split("/");
                String sn = split[5].toUpperCase();
                this.procRadarStatus(sn, content);
            } else if (topic.endsWith("thing/model/up_raw")) {
                log.warn("Unsupported topic:{}, content:{}", topic, content);
//                log.info("process topic:{}, content:{}", topic, content);
//                String[] split = topic.split("/");
//                String sn = split[3];
//                this.procUserData(sn, content);
            } else if (topic.endsWith("/thing/event/property/post")) {
                // 处理物模型上传数据
                log.info("process topic:{}, content:{} wlmxscsj", topic, content);
                String[] split = topic.split("/");
                String sn = split[2].toUpperCase();
                Map<String, Object> map = new HashMap<String, Object>();
                map.put("sn", sn);
                map.put("content", content);
                iotMessageHandleService.iotMessageHandle(map);
            } else {
                // log.warn("Unsupported topic:{}, content:{}", topic, content);
            }
        } catch (Exception e) {
            log.error("processMessage occurs error ", e);
        }
    }

    private void procRadarStatus(String sn, String content) {
//        JsonNode rootNode = JsonUtils.stringToJson(content);
//        if (rootNode == null || rootNode.get("status") == null) {
//            log.error("procRadarStatus error sn:{}, content:{}", sn, content);
//            return;
//        }
//        int status = RadarMessage.STATUS_OFFLINE;
//        if (rootNode.get("status").asText().equals("online")) {
//            status = RadarMessage.STATUS_ONLINE;
//        }
        IotRadarStatusData iotRadarStatusData = JSON.parseObject(content, IotRadarStatusData.class);
        if (Objects.isNull(iotRadarStatusData) || StringUtils.isEmpty(iotRadarStatusData.getStatus())) {
            log.error("procRadarStatus error sn:{}, content:{}", sn, content);
            return;
        }
        RadarMessage radarMessage = new RadarMessage();
        radarMessage.setSn(sn);
        radarMessage.setType(RadarMessage.TYPE_DEVICE_STATUS);
        radarMessage.setData(iotRadarStatusData);
        radarMessage.setTimestamp(System.currentTimeMillis());
        log.info("发送离线或者上线消息{}", sn);
        messageProcService.procRadarStatus(radarMessage);
    }

    private JmsConnectionListener myJmsConnectionListener = new JmsConnectionListener() {
        /**
         * 连接成功建立。
         */
        @Override
        public void onConnectionEstablished(URI remoteURI) {
            log.info("onConnectionEstablished, remoteUri:{}", remoteURI);
        }

        /**
         * 尝试过最大重试次数之后，最终连接失败。
         */
        @Override
        public void onConnectionFailure(Throwable error) {
            log.error("onConnectionFailure, {}", error.getMessage());
        }

        /**
         * 连接中断。
         */
        @Override
        public void onConnectionInterrupted(URI remoteURI) {
            log.info("onConnectionInterrupted, remoteUri:{}", remoteURI);
        }

        /**
         * 连接中断后又自动重连上。
         */
        @Override
        public void onConnectionRestored(URI remoteURI) {
            log.info("onConnectionRestored, remoteUri:{}", remoteURI);
        }

        @Override
        public void onInboundMessage(JmsInboundMessageDispatch envelope) {
        }

        @Override
        public void onSessionClosed(Session session, Throwable cause) {
        }

        @Override
        public void onConsumerClosed(MessageConsumer consumer, Throwable cause) {
        }

        @Override
        public void onProducerClosed(MessageProducer producer, Throwable cause) {
        }
    };

    /**
     * 计算签名，password组装方法，请参见AMQP客户端接入说明文档。
     */
    private static String doSign(String toSignString, String secret, String signMethod) throws Exception {
        SecretKeySpec signingKey = new SecretKeySpec(secret.getBytes(), signMethod);
        Mac mac = Mac.getInstance(signMethod);
        mac.init(signingKey);
        byte[] rawHmac = mac.doFinal(toSignString.getBytes());
        return Base64.encodeBase64String(rawHmac);
    }

    /**
     * 清空阿里云堆积消息
     *
     * @param accessKey
     * @param accessSecret
     * @param iotInstanceId
     * @param groupId
     */
    private static void clearMessage(String accessKey, String accessSecret, String iotInstanceId, String groupId) {
        DefaultProfile profile = DefaultProfile.getProfile("cn-beijing", accessKey, accessSecret);
        /** use STS Token
         DefaultProfile profile = DefaultProfile.getProfile(
         "<your-region-id>",           // The region ID
         "<your-access-key-id>",       // The AccessKey ID of the RAM account
         "<your-access-key-secret>",   // The AccessKey Secret of the RAM account
         "<your-sts-token>");          // STS Token
         **/
        IAcsClient client = new DefaultAcsClient(profile);

        ResetConsumerGroupPositionRequest request = new ResetConsumerGroupPositionRequest();
        request.setIotInstanceId(iotInstanceId);
        request.setGroupId(groupId);

        try {
            ResetConsumerGroupPositionResponse response = client.getAcsResponse(request);
            System.out.println(new Gson().toJson(response));
        } catch (ServerException e) {
            e.printStackTrace();
        } catch (ClientException e) {
            System.out.println("ErrCode:" + e.getErrCode());
            System.out.println("ErrMsg:" + e.getErrMsg());
            System.out.println("RequestId:" + e.getRequestId());
        }

    }

}
